tensor([[[[[0.5100, 0.0000],
           [0.0000, 0.5000]],

          [[0.0000, 0.2700],
           [0.2700, 0.0000]]],


         [[[0.5100, 0.4900],
           [0.2700, 0.0000]],

          [[0.1100, 0.0600],
           [0.1600, 0.1800]]]],



        [[[[0.0000, 0.5200],
           [0.5100, 0.0000]],

          [[0.0000, 0.2700],
           [0.2700, 0.0000]]],


         [[[0.0000, 0.5100],
           [0.2700, 0.5200]],

          [[0.0500, 0.0200],
           [0.0900, 0.1200]]]],



        [[[[0.0000, 0.0000],
           [0.4900, 0.5000]],

          [[0.5100, 0.2600],
           [0.2600, 0.5100]]],


         [[[0.4900, 0.0000],
           [0.2600, 0.0000]],

          [[0.3500, 0.2200],
           [0.1000, 0.1300]]]],



        [[[[0.4900, 0.4800],
           [0.0000, 0.0000]],

          [[0.4900, 0.2000],
           [0.2000, 0.4900]]],


         [[[0.0000, 0.0000],
           [0.2000, 0.4800]],

          [[0.4900, 0.7000],
           [0.6400, 0.5600]]]]], grad_fn=<RoundBackward1>)